package mydb

import "github.com/jinzhu/gorm"

//默认的数据库
var db *gorm.DB

//默认的数据库路径
var dbPath = "data.db"

//Init 初始化配置
func Init(path string) (err error) {
	dbPath = path
	db = GetDB()
	return
}

//GetDB 在调用此函数前必须先设置路径
func GetDB() *gorm.DB {
	if db == nil {
		var err error
		db, err = gorm.Open("sqlite3", dbPath)
		if err != nil {
			panic("连接数据库失败" + err.Error())
		}
	}
	return db
}

//Close 关闭数据库
func Close() {
	db := GetDB()
	if db != nil {
		db.Close()
		db = nil
	}
}
